#pragma once
#include <iostream>
#include <memory>
#include <sys/epoll.h>
#include "Socket.hpp"
#include <string>
static const int gsize = 64;
class EpollServer
{
public:
    EpollServer(uint16_t port)
        : listensocket(std::make_unique<TcpSocket>()),
          _port(port)

    {
        listensocket->BuildListenSocketMethod(_port);
        _epfd = epoll_create(12); // 这里面的参数随便写
        if (_epfd < 0)
        {
            LOG(LogLevel::FATAL) << "epoll_create falied";
        }
        LOG(LogLevel::DEBUG) << " _epfd :" << _epfd;
        struct epoll_event re;
        re.events = EPOLLIN;
        re.data.fd = listensocket->SockFd(); // 保存这个以后可以根据不同类型的fd执行不同的函数
        int n = epoll_ctl(_epfd, EPOLL_CTL_ADD, listensocket->SockFd(), &re);
        (void)n;
    }
    void accepter()
    {
        // 如果是监听套接字就绪
        InetAddr client;
        int clientfd = listensocket->Accept(&client);
        // 这里链接成功的话。我们需要把套接字添加到我们的epoll模型里面
        struct epoll_event re;
        re.events = EPOLLIN;
        re.data.fd = clientfd;
            epoll_ctl(_epfd, EPOLL_CTL_ADD, clientfd, &re);
        LOG(LogLevel::DEBUG) << "我们收到一个链接fd为" << clientfd;
    }
    void sender(int fd)
    {

        // 这里我们是不完整的，因为我们确认我们收到的是一个完整的报文，我们的应用上层也没有
        char buffer[1024];
        ssize_t n = recv(fd, buffer, sizeof(buffer) - 1, 0);
        if (n > 0)
        {
            buffer[n] = 0;
            std::string echo = "Echo says#";
            echo += buffer;
            send(fd, echo.c_str(), echo.size(), 0);
        }
    }
    void handler(int n)
    {
        for (int i = 0; i < n; i++)
        {
            int fd = revs[i].data.fd;
            int event = revs[i].events;
            if (event & EPOLLIN)
            {
                // 到这里表示读时间就绪
                if (fd == listensocket->SockFd())
                {
                    accepter();
                }
                else
                {
                    sender(fd);
                }
            }
            if (event & EPOLLET)
            {
            }
        }
    }
    void Run()
    {
        int timeout = 1000; // 为o表述轮询 为-1表示阻塞等待
        while (true)
        {
            int n = epoll_wait(_epfd, revs, gsize, timeout); // 成功返回就绪的个数，如果返回的是0的话表示超时
            switch (n)
            {
            case 0:
                LOG(LogLevel::WARNING) << "time out";
                break;
            case -1:
                LOG(LogLevel::FATAL) << "wait failed";
            default:
                handler(n);
                break;
            }
        }
    }
    ~EpollServer()
    {
    }

private:
    std::unique_ptr<Socket> listensocket;
    uint16_t _port;
    int _epfd;
    struct epoll_event revs[gsize];
};